python code to loop through two lists